from rest_framework import viewsets

from apps.project.models import ProjectModel, ProjectUserModel
from apps.project.serializers import ProjectSerializer, ProjectUserSerializer


class ProjectViewSet(viewsets.ModelViewSet):
    queryset = ProjectModel.objects.all()
    serializer_class = ProjectSerializer


class ProjectUserViewSet(viewsets.ModelViewSet):
    queryset = ProjectUserModel.objects.all()
    serializer_class = ProjectUserSerializer
